tensor([[[0.3000, 0.3000],
         [0.3000, 0.3000]],

        [[0.2700, 0.2700],
         [0.2700, 0.2700]],

        [[0.4300, 0.4300],
         [0.4300, 0.4300]]], grad_fn=<RoundBackward1>)